package com.baidu.leetcode.hashtable;

import java.util.Scanner;

/**
 * @author shilei
 * @create 2021-12-10 18:18
 * 使用 hash 数据结构演示雇员信息的管理
 */
public class HashTableDemo {
    public static void main(String[] args) {
        HashTab hashTab = new HashTab(7);
        //写一个简单菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:查找雇员");
            System.out.println("exit:退出");
            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    //创建雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("请输入要查找的id");
                    id =scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                    break;
            }
        }

    }
}

//创建hashTab  管理多条链表
class HashTab {
    private EmpLinkedList[] empLinkedList;
    private int size;//表示有多少条链表

    public HashTab(int size) {
        this.size = size;
        this.empLinkedList = new EmpLinkedList[size];
        //这时不要忘了分别初始化每个链表
        for (int i = 0; i < size; i++) {
            empLinkedList[i] = new EmpLinkedList();
        }
    }

    //添加雇员
    public void add(Emp emp) {
        //根据员工的ID 得到该员工应当添加到那条链表
        int empLinkedListNo = hashFun(emp.id);
        //将 emp 加入到对应的链表中
        empLinkedList[empLinkedListNo].add(emp);
    }

    //遍历
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedList[i].list(i);
        }
    }

    //根据id查找雇员
    public void findEmpById(int id){
        //使用散列函数 确定去哪个链表查找
        int empNo = hashFun(id);
        Emp empById = empLinkedList[empNo].findEmpById(id);
       if (null != empById) {
           System.out.printf("在第%d条链表中找到 雇员 id=%d\n",empNo,id);
       }else {
           System.out.println("当前雇员不存在");
       }
    }

    //编写散列函数
    public int hashFun(int id) {
        return id % size;
    }
}

//表示一个雇员信息
class Emp {
    public int id;
    public String name;
    public Emp next;//next 默认为空

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

//创建链表
class EmpLinkedList {
    //头指针 执行第一个Emp 因此我们这个链表的head  是直接指向第一个Emp
    private Emp head;//默认null

    //添加雇员到链表
    /*
    说明：
        1、假定 当添加雇员时 id是自增 即id分配总是从小到大
        2、因此我们将该雇员直接加入到本链表的最后即可

     */
    public void add(Emp emp) {
        //如果是添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是添加第一个雇员 则使用一个辅助的指针 帮助定位到最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.next == null) {//说明到链表最后
                break;
            }
            curEmp = curEmp.next;//后移
        }
        //退出时直接将emp 加入链表
        curEmp.next = emp;
    }

    //遍历链表的雇员信息
    public void list(int index) {
        if (head == null) {
            //说明链表为空
            System.out.println("当前链表为空,链表编号为:" + index);
            return;
        }
        System.out.println(index + " 当前链表的信息为");
        Emp curEmp = head;//辅助指针
        while (true) {
            System.out.printf("=> id=%d name=%s\t", curEmp.id, curEmp.name);
            if (curEmp.next == null) {
                //说明curEmp是最后节点
                break;
            }
            curEmp = curEmp.next;//后移遍历
        }
        System.out.println();
    }

    //根据id查找雇员
    public Emp findEmpById(int id) {
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        Emp curEmp = head;
        while (true) {
            if (curEmp.id == id) {
                return curEmp;
            }
            if (curEmp.next == null) {
                //说明链表不存在当前雇员
                return null;
            }
            curEmp = curEmp.next;
        }
    }
}